package cn.ycc1.my.agent.config;

import dev.langchain4j.model.openai.OpenAiStreamingChatModel;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

/**
 * @author ycc
 * @date 2025/3/1
 */
@Configuration
public class LlmConfig {
    @Bean
    public OpenAiStreamingChatModel openAiStreamingChatModel() {

//        return OpenAiStreamingChatModel.builder()
//                .apiKey(System.getenv("DEEPSEEK_API_KEY"))
//                .baseUrl("https://api.deepseek.com/v1")
//      .modelName("deepseek-reasoner")
//      .temperature(0.7)
//      .logRequests(true)
//      .logResponses(true)
//      .build();

        return OpenAiStreamingChatModel.builder()
                .apiKey(System.getenv("DASHSCOPE_API_KEY"))
                .baseUrl("https://dashscope.aliyuncs.com/compatible-mode/v1")
                .modelName("deepseek-r1")
                .temperature(0.7)
                .logRequests(true)
                .logResponses(true)
                .build();
    }

//    @Bean
//    OpenAiChatModel openAiChatModel() {
//
//        return OpenAiChatModel.builder()
//                .apiKey(System.getenv("DEEPSEEK_API_KEY"))
//                .baseUrl("https://api.deepseek.com/v1")
//                .modelName("deepseek-reasoner")
//                .temperature(0.7)
//                .logRequests(true)
//                .logResponses(true)
//                .build();
//    }
}
